MODEL_CONFIGS = {
    'msl': {
        'model_path': 'data/pretrained/msl/best_05_03-16_21_16.pt',
        'dataset': 'msl',
        'train_config': {
            'slide_win': 5,
            'slide_stride': 1,
            'batch': 32,
            'dim': 64,
            'out_layer_num': 1,
            'out_layer_inter_dim': 128,
            'topk': 5
        }
    },
    # 可以添加更多模型的配置
}


def get_model_config(model_name):
    """
    获取模型配置
    Args:
        model_name: 模型名称
    Returns:
        tuple: (model_path, dataset, train_config)
    """
    if model_name not in MODEL_CONFIGS:
        raise ValueError(f"未找到模型 '{model_name}' 的配置，可用的模型有: {list(MODEL_CONFIGS.keys())}")

    config = MODEL_CONFIGS[model_name]
    return config['model_path'], config['dataset'], config['train_config']